import pandas as pd
# 读取excel1的内容到excel2
#
# method1
# dataFrame = pd.read_excel('D:\\WILNEVKL\\勘察设计关键数据表--室分.xls', sheet_name='Sheet1')#, index_col=[2, 4]
# dataFrame[['地市','区县']].to_excel('D:\\WILNEVKL\\新建.xls', sheet_name='Sheet1')
# method2
# dataFrame = pd.read_excel('D:\\WILNEVKL\\勘察设计关键数据表--室分.xls', sheet_name='Sheet1',usecols=[2,3,4])#, index_col=[2, 4]
# dataFrame.to_excel('D:\\WILNEVKL\\新建5.xls', sheet_name='Sheet1', startcol=2, index=False )
# method3 手动输入文件路径
# print('2,3,4'.split(','))

inputFile = input("源文件:")
print('源文件', inputFile)
outputFile = input("目标文件:")
print('目标文件：', outputFile)
copyClm = input("复制源文件的哪几列:")
# print('从源文件的第几列开始复制：', inputFile)
copyClm = copyClm.split(',')
# method1转换元素为int
for index in range(len(copyClm)):
   copyClm[index]=int(copyClm[index])-1
print(copyClm)
copyToIndex = int(input("复制到目标文件，从第几列开始:"))+1
# print('input：', copyToIndex)
# method2 转换元素为int
# indexClm =[]
# print(type(copyClm))
# for idx in copyClm:
#     indexClm.append(int(idx))
dataFrame = pd.read_excel(inputFile, sheet_name='Sheet1',usecols=copyClm)
# dataFrame = pd.read_excel(inputFile, sheet_name='Sheet1',usecols=(2,3,4))#, 获取第2,3,4列（从第0列开始计数）进行复制
dataFrame.to_excel(outputFile, sheet_name='Sheet1', startcol=int(copyToIndex), index=False) #从第2列开始写入
# dataFrame.to_excel('D:\\WILNEVKL\\新建.xls', sheet_name='Sheet1')
inputFile = input("输入任意退出:")
